#pragma once

#include <iostream>
#include <queue>
#include <pthread.h>
#include <unistd.h>
#include <cstdlib>
#include <ctime>

#define NUM 6

template <class T>
class PC
{
public:
    PC(int cap = NUM) : _cap(cap)
    {
        // 初始化锁和条件变量
        pthread_mutex_init(&lock, nullptr);
        pthread_cond_init(&_empty, nullptr);
        pthread_cond_init(&_full, nullptr);
    }

    void Push(const T &in)
    {
        // 先申请锁
        pthread_mutex_lock(&lock);
        // 判断是否为满
        while (full())
        {
            // 如果是满了，那么就必须等待，并把锁释放出去
            pthread_cond_wait(&_full, &lock);
        }
        // 插入数据
        q.push(in);
        // 队列中已经有数据了，唤醒正在等待的消费者。
        if (q.size() >= _cap / 2)
        {
            pthread_cond_signal(&_empty);
            std::cout << "消费者快来消费吧  ";
        }
        // 释放锁
        pthread_mutex_unlock(&lock);
    }
    void Pop(T &out)
    {
        // 先申请锁
        pthread_mutex_lock(&lock);
        // 判断队列是否为空
        while (empty())
        {
            // 如果为空，那么消费者等待，并释放锁
            pthread_cond_wait(&_empty, &lock);
        }
        // 拿出队头的资源
        out = q.front();
        // 删除队头数据
        q.pop();
        // 唤醒正在等待的生产者
        if (q.size() <= _cap / 2)
        {
            pthread_cond_signal(&_full);
            std::cout << "生产者快来生产吧  ";
        }
        // 释放锁
        pthread_mutex_unlock(&lock);
    }
    ~PC()
    {
        _cap = 0;
        // 销毁锁和条件变量
        pthread_mutex_destroy(&lock);
        pthread_cond_destroy(&_full);
        pthread_cond_destroy(&_empty);
    }

private:
    // 判断队列是否为空
    bool empty()
    {
        return q.empty();
    }
    // 判断队列是否为满
    bool full()
    {
        return q.size() == _cap;
    }

private:
    std::queue<T> q;
    int _cap;
    // 定义锁
    pthread_mutex_t lock;
    // 定义条件变量，_empty表示队列为空的条件变量,_full表示队列为满的条件变量
    pthread_cond_t _empty;
    pthread_cond_t _full;
};
